package com.shiyun.disaster_station.config;

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
import com.shiyun.disaster_station.wrapper.UpdateBatchMethod;

import java.util.List;

/**
 * 类 描 述: 添加自定SQL注入器
 * 作   者: 史云
 *
 * @author 史云
 * 时   间: 2022年02月28日  14:32:00
 */
public class CustomizedSqlInjector extends DefaultSqlInjector {

    /**
     * 方法描述: 如果只需增加方法，保留mybatis plus自带方法，
     * 可以先获取super.getMethodList()，再添加add
     * 作   者: 史 云
     * 时   间: 2022年02月28日 14:33:00
     */
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass,tableInfo);
        methodList.add(new InsertBatchSomeColumn());
        methodList.add(new UpdateBatchMethod());
        return methodList;
    }
}


